import unittest
from htmltestreport import HTMLTestReport

from common.login import init_log_config
from config import BASE_DIR

if __name__ == '__main__':
    init_log_config(BASE_DIR+'/log/ihrm.log')

    # - 需要通过命令运行代码，才能生成测试报告
    # 2. 生成测试套件
    suite = unittest.defaultTestLoader.discover(BASE_DIR + '/scripts', 'test*')

    # 3. 实例化HTMLTestRunner对象
    report = HTMLTestReport(BASE_DIR+'/report/report.html',
                            title="IHRM测试报告",
                            description="V1.0")

    # 4. 执行测试套件
    report.run(suite)
